counting in python